package com.lzh.java;

/**
 * @author liuzhihao
 * @create 2021-02-19 17:46
 */

import org.junit.Test;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;

/**
 * 集合元素的遍历操作，使用迭代器Iterator接口
 *  1.内部的方法：hasNext和next()
 *  2.集合对象每次调用iterator()方法都得到一个全新的迭代器对象，默认游标都在集合的第一个元素之前。
 *  3.内部定义了remove(),可以在遍历的时候，删除集合中的元素。此方法不同于集合直接调用remove()
 */
public class IteratorTest {

    @Test
    public void test1(){
        Collection coll = new ArrayList();
        coll.add(123);
        coll.add(456);
        coll.add(new Person("liuzhihao",22));
        coll.add(new String("Tom"));
        coll.add(false);

        Iterator it = coll.iterator();

        //方式一：
//        System.out.println(it.next());
//        System.out.println(it.next());
//        System.out.println(it.next());
//        System.out.println(it.next());
//        System.out.println(it.next());
//        //报异常：NoSuchElementException
//        System.out.println(it.next());

        //方式二
//        for(int i = 0;i<coll.size();i++){
//            System.out.println(it.next());
//        }

        //方式三：推荐
        //hasNext():判断是否还有下一个元素
        while(it.hasNext()){
            ///next():①指针下移 ②将下移以后集合位置上的元素返回
            System.out.println(it.next());
        }
    }

    @Test
    public void test2(){
        Collection coll = new ArrayList();
        coll.add(123);
        coll.add(456);
        coll.add(new Person("liuzhihao",22));
        coll.add(new String("Tom"));
        coll.add(false);

//        Iterator it = coll.iterator();
        //错误方式一：
//        while(it.next() != null){
//            System.out.println(it.next());
//        }

        //错误方式二：
        //集合对象每次调用iterator()方法都得到一个全新的迭代器对象，默认游标都在集合
        //的第一个元素之前。
        while(coll.iterator().hasNext()){
            System.out.println(coll.iterator().next());
        }

    }

    //测试Iterator中的remove()
    // 如果还未调用next()或在上一次调用 next 方法之后已经调用了 remove 方法，
    //再调用remove都会报IllegalStateException。
    @Test
    public void test3(){
        Collection coll = new ArrayList();
        coll.add(123);
        coll.add(456);
        coll.add(new Person("liuzhihao",22));
        coll.add(new String("Tom"));
        coll.add(false);

        //删除“Tom”对象
        Iterator it = coll.iterator();
        while(it.hasNext()){
            Object obj = it.next();
            if("Tom".equals(obj)){
                it.remove();
            }
        }

        it = coll.iterator();
        while(it.hasNext()){
            System.out.println(it.next());
        }

    }
}
